package com.dragon.utils;


import java.util.regex.Matcher;
import java.util.regex.Pattern;

public class GetClassNameUtils {
    /**
     * 用来获取代码中的主类名
     * @param code 用户提交的代码
     * @return 提取的类名
     */
    public static String getClassNameUtils(String code) {
        Pattern pattern = Pattern.compile("public\\s+class\\s+([A-Za-z_][A-Za-z0-9_]*)");
        Matcher matcher = pattern.matcher(code);

        if(matcher.find()) {
            // group(1) 返回第一个捕获的内容
            return matcher.group(1);
        }
        return "Main";
    }
}
